ECS RunTaskで実行されるSpring Batchアプリケーションにおいて、特定のエラー条件時のみリトライさせてみる

@Component("FileDownloadTasklet")
@StepScope
@Slf4j
public class FileDownloadTasklet implements Tasklet {

  @Override
  public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
    StepExecution stepExecution = chunkContext.getStepContext().getStepExecution();
    S3Client s3Client = S3Client.builder()
        .region(Region.AP_NORTHEAST_1)
        .build();
    String bucketName = System.getenv("BUCKET_NAME");
    String fileKey = System.getenv("FILE_KEY");
    try {
      GetObjectRequest getObjectRequest = GetObjectRequest.builder()
          .bucket(bucketName)
          .key(fileKey)
          .build();
      s3Client.getObject(getObjectRequest);
    } catch (S3Exception e) {
       log.error(e.awsErrorDetails().errorMessage());
      if (e.awsErrorDetails().errorCode().equals("NoSuchKey")) {
        log.error("NoSuchKey");
        stepExecution.setStatus(BatchStatus.ABANDONED);
      } else {
        stepExecution.setStatus(BatchStatus.FAILED);
      }
    }

    return RepeatStatus.FINISHED;
  }

}

def is_retryable(cause):
    # Retryable errors ECS returns
    # https://docs.aws.amazon.com/AmazonECS/latest/userguide/stopped-task-error-codes.html
    if cause['StoppedReason'].startswith("CannotPullContainerError"):
        return True
    if cause['StoppedReason'].startswith("ResourceInitializationError"):
        return True

    # Retryable errors your application returns
    contianer = cause['Containers'][0]
    if contianer.get('ExitCode', 0) in [6]:
        return True
    return False